#include <iostream>
#include <vector>
#include <map>
#include <ctime>

#define CEHUA 0
#define MEISHU 1
#define YANFA 2

using namespace std;
//员工分组
//案例描述
//1、公司今天招聘了10个员工（ABCDEFGHIJ），10名员工进入公司之后，需要指派员工在那个部门工作
//2、员工信息有: 姓名  工资组成；部门分为：策划、美术、研发
//3、随机给10名员工分配部门和工资
//4、通过multimap进行信息的插入  key(部门编号) value(员工)
//5、分部门显示员工信息

//实现步骤
//1. 创建10名员工，放到vector中
//2. 遍历vector容器，取出每个员工，进行随机分组
//3. 分组后，将员工部门编号作为key，具体员工作为value，放入到multimap容器中
//4. 分部门显示员工信息

class Worker {
public:
    string name;
    int salary;
};

void createWorker(vector<Worker> &v) {
    string nameSeed = "ABCDEFGHIJ";
    for (int i = 0; i < 10; ++i) {
        Worker worker;
        worker.name = "员工";
        worker.name += nameSeed[i];

        worker.salary = rand() % 10000 + 10000;//10000~19999
        v.push_back(worker);//将员工放入到容器中
    }
}

void setGroup(vector<Worker> &v, multimap<int, Worker> &m) {
    for (vector<Worker>::iterator it = v.begin(); it != v.end(); ++it) {
        //产生随机编号
        int id = rand() % 3;//0 1 2
        m.insert(make_pair(id, *it));//将员工插入到分组中 key表示部门编号 value表示具体员工
    }
}

void show(multimap<int, Worker> &m) {
    cout << "策划部门:" << endl;
    multimap<int, Worker>::iterator pos = m.find(CEHUA);
    int count = m.count(CEHUA);
    int index;
    for (index = 0; pos != m.end() && index < count; ++pos, ++index) {
        cout << "name:" << pos->second.name << ",salary:" << pos->second.salary << endl;
    }

    cout << "=========" << endl;

    cout << "美术部门:" << endl;
    pos = m.find(MEISHU);
    count = m.count(MEISHU);;
    for (index = 0; pos != m.end() && index < count; ++pos, ++index) {
        cout << "name:" << pos->second.name << ",salary:" << pos->second.salary << endl;
    }

    cout << "=========" << endl;

    cout << "研发部门:" << endl;
    pos = m.find(YANFA);
    count = m.count(YANFA);;
    for (index = 0; pos != m.end() && index < count; ++pos, ++index) {
        cout << "name:" << pos->second.name << ",salary:" << pos->second.salary << endl;
    }
}

int main() {
    srand((unsigned int) time(nullptr));
    //1、创建员工
    vector<Worker> v;
    createWorker(v);

    //2、员工分组
    multimap<int, Worker> m;
    setGroup(v, m);

    //3、分组显示员工
    show(m);

    //测试
//    cout<<"========="<<endl;
//    for (vector<Worker>::iterator it = v.begin(); it != v.end(); ++it) {
//        cout << "name:" << it->name << ",salary:" << it->salary << endl;
//    }
    return 0;
}